# -*- coding: utf-8 -*-
# @Time : 2021-11-17 18:49
# @Author : lwb
# @Site : 
# @File : dataset.py
from torch.utils.data import DataLoader, Dataset
# 对于dataset，一般只需要实现以下三个三个即可
class BowDataset(Dataset):
    def __init__(self, data):  # data为原始训练
        self.data = data

    def __len__(self):  # 返回数据集中杨烈的个数
        return len(self.data)

    def __getitem__(self, i):  # 返回索引值为item的样本
        return self.data[i]
